CUDA_VISIBLE_DEVICES=0 python train.py \
--dataset_name pubmed_refind_28w \
--dir_output pubmed_refind_28w_1 